<!DOCTYPE html>
<html>
<!--
Copyright 2008 The Closure Library Authors. All Rights Reserved.

Use of this source code is governed by an Apache 2.0 License.
See the COPYING file for details.
-->
<!--

  Author: nkanodia@google.com
-->
<head>
<title>Closure Unit Tests - goog.testing.MockControl</title>
<script src="../base.js"></script>
<script>
  goog.require('goog.testing.MockControl');
  goog.require('goog.testing.jsunit');
</script>
</head>
<body>
<script>

  // Emulate the behavior of a mock.
  function MockMock() {
    this.replayCalled = false;
    this.resetCalled = false;
    this.verifyCalled = false;
    this.tearDownCalled = false;
  };

  MockMock.prototype.$replay = function() {
    this.replayCalled = true;
  };

  MockMock.prototype.$reset = function() {
    this.resetCalled = true;
  };

  MockMock.prototype.$verify = function() {
    this.verifyCalled = true;
  };

  function setUp() {
    var mock = new goog.testing.Mock(MockMock);
  };

  function testAdd() {
    var mockMock = new MockMock();

    var control = new goog.testing.MockControl();
    assertEquals(mockMock, control.addMock(mockMock));
  }

  function testReplayAll() {
    var mockMock1 = new MockMock();
    var mockMock2 = new MockMock();
    var mockMockExcluded = new MockMock();

    var control = new goog.testing.MockControl();
    control.addMock(mockMock1);
    control.addMock(mockMock2);

    control.$replayAll();
    assertTrue(mockMock1.replayCalled);
    assertTrue(mockMock2.replayCalled);
    assertFalse(mockMockExcluded.replayCalled);
  }

  function testResetAll() {
    var mockMock1 = new MockMock();
    var mockMock2 = new MockMock();
    var mockMockExcluded = new MockMock();

    var control = new goog.testing.MockControl();
    control.addMock(mockMock1);
    control.addMock(mockMock2);

    control.$resetAll();
    assertTrue(mockMock1.resetCalled);
    assertTrue(mockMock2.resetCalled);
    assertFalse(mockMockExcluded.resetCalled);
  }

  function testVerifyAll() {
    var mockMock1 = new MockMock();
    var mockMock2 = new MockMock();
    var mockMockExcluded = new MockMock();

    var control = new goog.testing.MockControl();
    control.addMock(mockMock1);
    control.addMock(mockMock2);

    control.$verifyAll();
    assertTrue(mockMock1.verifyCalled);
    assertTrue(mockMock2.verifyCalled);
    assertFalse(mockMockExcluded.verifyCalled);
  }

  function testTearDownAll() {
    var mockMock1 = new MockMock();
    var mockMock2 = new MockMock();
    var mockMockExcluded = new MockMock();

    // $tearDown is optional.
    mockMock2.$tearDown = function() {
      this.tearDownCalled = true;
    };
    mockMockExcluded.$tearDown = function() {
      this.tearDownCalled = true;
    };

    var control = new goog.testing.MockControl();
    control.addMock(mockMock1);
    control.addMock(mockMock2);

    control.$tearDown();

    // mockMock2 has a tearDown method and is in the control.
    assertTrue(mockMock2.tearDownCalled);
    assertFalse(mockMock1.tearDownCalled);
    assertFalse(mockMockExcluded.tearDownCalled);
  }

</script>
</body>
</html>
